from models.dnns import DNNs
import os
import torch
import moxing as mox

# save_path = r'../save/dnns_0'
# if not os.path.exists(save_path):
#     os.makedirs(save_path)

pth_root_path = r'../save/dnn0_3layers'
pth_list = [
    r'',
    r'',
    r'',
    r'',
    r''
]
for i,pth_name in enumerate(pth_list):
    pth_list[i] = os.path.join(pth_root_path,pth_name)
    if not os.path.exists(pth_list[i]):
        import

model = DNNs(pth_list)
model.eval()
save_path = os.path.join(pth_root_path,'DNNs.pth')
torch.save(model,save_path)
model_path = 'obs://ad-competiton/my_baseline/model'

mox.file.copy(save_path, model_path + '/std.npy')
